import tensorflow as tf 
from tensorflow import keras

strategy = tf.distribute.MirroredStrategy()  # 定义分布式策略

config = tf.estimator.RunConfig(train_distribute=strategy)  # 创建训练设置

estimator = keras.estimator.model_to_estimator(model,config=config)  # keras转换时加入

# tf2 推荐使用keras转换成estimator  不推荐使用预先设置的
# tf1的estimator的分布式还要查资料